package tmhprediction.classification;


public class SecondClassifier {

	//Results of grid search
	private double c = Math.pow(2.0,-1.0);			// TODO these values are NOT the final ones	
	private double g = Math.pow(2.0,-7.0);
	
	private int classifierCacheSizeInMB;
	
	TMHClassifier classifierWrapper;
	
	public SecondClassifier()
	{
		this.classifierCacheSizeInMB = 40;
	}
	public SecondClassifier(int pCacheSize)
	{
		this.classifierCacheSizeInMB = pCacheSize;
	}
	
	public void initializeTrainedClassifier(String pModelFile, String pTestSetFile) throws Exception
	{
		this.classifierWrapper = new TMHClassifier(pModelFile,pTestSetFile);
		
	}
	
	public void initializeUntrainedClassifier(String pTrainingArffFile, String pTestArffFile) throws Exception
	{
		this.classifierWrapper = new TMHClassifier(pTrainingArffFile, pTestArffFile, this.g, this.c, classifierCacheSizeInMB);
	}
	
	public TMHClassifier getClassifierWrapper()
	{
		return this.classifierWrapper;
	}

}
